package com.seek.sms.controller;

import com.seek.commonutils.ResponseResult;
import com.seek.sms.service.SmsService;
import com.seek.sms.utils.RandomUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.*;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * @author: wml
 * @since: 2022/2/19 14:16
 */
@RestController
@RequestMapping("/edusms/sms")
public class SmsController {

    @Autowired
    private SmsService smsService;

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    // 发送短信的方法（阿里云短信服务）
    @GetMapping("send/{phone}")
    public ResponseResult sendSms(@PathVariable String phone) {
        // 从 redis 中获取验证码，如果获取到直接返回
        String code = redisTemplate.opsForValue().get(phone);
        if (code != null && !code.equals("")) {
            return ResponseResult.ok();
        }

        // 如果 redis 获取不到，进行阿里云发送
        // 生成随机值，传递给阿里云进行发送
        code = RandomUtil.getSixBitRandom();
        Map<String, Object> param = new HashMap<>();
        param.put("code", code);
        // 调用 service 中发送短信的方法
        boolean isSend = smsService.sendSms(phone, param);
        if (isSend) {
            // 发送成功，把发送成功的验证码放到 redis 里面并设置有效时间
            redisTemplate.opsForValue().set(phone, code, 5, TimeUnit.MINUTES);
            return ResponseResult.ok();
        } else {
            return ResponseResult.error().message("短信发送失败");
        }
    }

    // 发送短信的方法（阿里云106短信服务）
    @GetMapping("send106/{phone}/{status}")
    public ResponseResult send106Sms(@PathVariable String phone, @PathVariable String status) {
        if (phone.equals("17822000246")) {
            return ResponseResult.ok();
        }
        // 从 redis 中获取验证码，如果获取到直接返回
        String code = null;
        if (status.equals("0")) {
            code = redisTemplate.opsForValue().get("register-" + phone);
        } else if (status.equals("1")) {
            code = redisTemplate.opsForValue().get("login-" + phone);
        } else {
            code = redisTemplate.opsForValue().get("forgetPassword-" + phone);
        }

        if (code != null && !code.equals("")) {
            return ResponseResult.ok();
        }

        // 如果 redis 获取不到，进行阿里云发送
        // 生成随机值，传递给阿里云进行发送
        code = RandomUtil.getSixBitRandom();
        // 调用 service 中发送短信的方法
        boolean isSend = smsService.send106Sms(phone, code);
        if (isSend) {
            // 发送成功，把发送成功的验证码放到 redis 里面并设置有效时间
            if (status.equals("0")) {
                redisTemplate.opsForValue().set("register-" + phone, code, 5, TimeUnit.MINUTES);
            } else if (status.equals("1")) {
                redisTemplate.opsForValue().set("login-" + phone, code, 5, TimeUnit.MINUTES);
            } else {
                redisTemplate.opsForValue().set("forgetPassword-" + phone, code, 5, TimeUnit.MINUTES);
            }
            return ResponseResult.ok();
        } else {
            return ResponseResult.error().message("短信发送失败");
        }
    }
}
